import numpy as np
nd_array = np.arange(25).reshape(5,5)
print(nd_array)
print('-'*35)
rows_array = np.arange(5)
print(nd_array[rows_array == 2, 2:])